cmake_minimum_required(VERSION 3.18)
project(torch_cpp C CXX)

set(CMAKE_CXX_STANDARD 17)

list(APPEND CMAKE_PREFIX_PATH /home/madengyun/miniconda3/envs/pytorch/lib/python3.10/site-packages)
# set(Torch_ROOT /home/madengyun/MyGit/pytorch/torch)
find_package(Torch REQUIRED)

add_executable(main torch_cpp_demo.cpp)
target_link_libraries(main PUBLIC ${TORCH_LIBRARIES})
